"""
AI智能助手后端示例

这个文件提供了一个完整的后端服务示例，展示如何：
1. 创建一个简单的Web服务器
2. 接收前端请求
3. 调用DeepSeek API
4. 返回AI响应

注意：要运行此服务，需要先安装依赖：
pip install openai flask

然后设置环境变量：
set DEEPSEEK_API_KEY=你的API密钥
"""

from flask import Flask, request, jsonify, send_from_directory
import os

# 这里提供完整的代码，实际运行时取消注释
'''
from openai import OpenAI

# 初始化OpenAI客户端
client = OpenAI(
    api_key=os.environ.get('DEEPSEEK_API_KEY'),
    base_url="https://api.deepseek.com"
)
'''

app = Flask(__name__)

@app.route('/')
def serve_index():
    """提供前端HTML文件"""
    return send_from_directory('.', 'index.html')

@app.route('/api/chat', methods=['POST'])
def chat():
    """处理聊天请求，调用DeepSeek API"""
    try:
        data = request.json
        user_message = data.get('message', '')
        
        if not user_message:
            return jsonify({'error': '消息不能为空'}), 400
        
        # 模拟AI响应 - 实际使用时替换为真实API调用
        # 以下是真实API调用的代码示例
        '''
        response = client.chat.completions.create(
            model="deepseek-chat",
            messages=[
                {"role": "system", "content": "你是一个有帮助的AI助手"},
                {"role": "user", "content": user_message},
            ],
            stream=False
        )
        
        ai_response = response.choices[0].message.content
        ''',
        
        # 模拟响应
        mock_responses = [
            f"感谢你的提问：{user_message}！这是一个很好的话题。",
            f"关于'{user_message}'，我有以下建议...",
            f"你提出了一个有趣的问题'{user_message}'，让我思考一下...",
            f"我理解你对'{user_message}'的疑问，以下是我的解答。",
            f"'{user_message}'是一个值得探讨的话题，我的看法是..."
        ]
        import random
        ai_response = random.choice(mock_responses)
        
        return jsonify({
            'success': True,
            'response': ai_response
        })
        
    except Exception as e:
        print(f"发生错误: {str(e)}")
        return jsonify({
            'success': False,
            'error': str(e)
        }), 500

@app.route('/health')
def health_check():
    """健康检查端点"""
    return jsonify({'status': 'healthy'})

if __name__ == '__main__':
    print("AI助手后端服务启动中...")
    print("注意：当前使用模拟数据响应，要使用真实API请取消相关代码注释")
    print("请确保已设置DEEPSEEK_API_KEY环境变量")
    print("服务运行在 http://localhost:5000")
    
    # 运行开发服务器
    app.run(host='0.0.0.0', port=5000, debug=True)